/*
 * Copyright © 2021 Google, Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice (including the next
 * paragraph) shall be included in all copies or substantial portions of the
 * Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#ifdef X
#  undef X
#endif

#if PTRSZ == 32
#  define X(n) n ## _32
#else
#  define X(n) n ## _64
#endif


static void
X(emit_reloc_common)(struct fd_ringbuffer *ring, const struct fd_reloc *reloc)
{
	(*ring->cur++) = (uint32_t)reloc->iova;
#if PTRSZ == 64
	(*ring->cur++) = (uint32_t)(reloc->iova >> 32);
#endif
}

static void
X(msm_ringbuffer_sp_emit_reloc_nonobj)(struct fd_ringbuffer *ring,
		const struct fd_reloc *reloc)
{
	X(emit_reloc_common)(ring, reloc);

	assert(!(ring->flags & _FD_RINGBUFFER_OBJECT));

	struct msm_ringbuffer_sp *msm_ring = to_msm_ringbuffer_sp(ring);

	struct msm_submit_sp *msm_submit =
			to_msm_submit_sp(msm_ring->u.submit);

	msm_submit_append_bo(msm_submit, reloc->bo);
}

static void
X(msm_ringbuffer_sp_emit_reloc_obj)(struct fd_ringbuffer *ring,
		const struct fd_reloc *reloc)
{
	X(emit_reloc_common)(ring, reloc);

	assert(ring->flags & _FD_RINGBUFFER_OBJECT);

	struct msm_ringbuffer_sp *msm_ring = to_msm_ringbuffer_sp(ring);

	/* Avoid emitting duplicate BO references into the list.  Ringbuffer
	 * objects are long-lived, so this saves ongoing work at draw time in
	 * exchange for a bit at context setup/first draw.  And the number of
	 * relocs per ringbuffer object is fairly small, so the O(n^2) doesn't
	 * hurt much.
	 */
	bool found = false;
	for (int i = 0; i < msm_ring->u.nr_reloc_bos; i++) {
		if (msm_ring->u.reloc_bos[i] == reloc->bo) {
			found = true;
			break;
		}
	}
	if (!found) {
		APPEND(&msm_ring->u, reloc_bos, fd_bo_ref(reloc->bo));
	}
}

static uint32_t
X(msm_ringbuffer_sp_emit_reloc_ring)(struct fd_ringbuffer *ring,
		struct fd_ringbuffer *target, uint32_t cmd_idx)
{
	struct msm_ringbuffer_sp *msm_target = to_msm_ringbuffer_sp(target);
	struct fd_bo *bo;
	uint32_t size;

	if ((target->flags & FD_RINGBUFFER_GROWABLE) &&
			(cmd_idx < msm_target->u.nr_cmds)) {
		bo   = msm_target->u.cmds[cmd_idx].ring_bo;
		size = msm_target->u.cmds[cmd_idx].size;
	} else {
		bo   = msm_target->ring_bo;
		size = offset_bytes(target->cur, target->start);
	}

	if (ring->flags & _FD_RINGBUFFER_OBJECT) {
		X(msm_ringbuffer_sp_emit_reloc_obj)(ring, &(struct fd_reloc){
			.bo     = bo,
			.iova   = bo->iova + msm_target->offset,
			.offset = msm_target->offset,
		});
	} else {
		X(msm_ringbuffer_sp_emit_reloc_nonobj)(ring, &(struct fd_reloc){
			.bo     = bo,
			.iova   = bo->iova + msm_target->offset,
			.offset = msm_target->offset,
		});
	}

	if (!(target->flags & _FD_RINGBUFFER_OBJECT))
		return size;

	struct msm_ringbuffer_sp *msm_ring = to_msm_ringbuffer_sp(ring);

	if (ring->flags & _FD_RINGBUFFER_OBJECT) {
		for (unsigned i = 0; i < msm_target->u.nr_reloc_bos; i++) {
			APPEND(&msm_ring->u, reloc_bos, fd_bo_ref(msm_target->u.reloc_bos[i]));
		}
	} else {
		// TODO it would be nice to know whether we have already
		// seen this target before.  But hopefully we hit the
		// append_bo() fast path enough for this to not matter:
		struct msm_submit_sp *msm_submit = to_msm_submit_sp(msm_ring->u.submit);

		for (unsigned i = 0; i < msm_target->u.nr_reloc_bos; i++) {
			msm_submit_append_bo(msm_submit, msm_target->u.reloc_bos[i]);
		}
	}

	return size;
}
